#pragma once

#include <iostream>
#include "Socket.hpp"
#include "Connection.hpp"

using namespace socket_ns;

class Listener
{
public:
    Listener(uint16_t port): _port(port), _listensock(std::make_unique<TcpSoket>())
    {
        _listensock->BuildListenSocket(_port);
    }
    
    int GetSockfd()
    {
        return _listensock->Sockfd();
    }

    // 处理新到来的连接
    void Accepter(Connection *con)
    {
        LOG(DEBUG, "新连接到来，准备接收\n");
        // 新到来的不一定只有一个连接，所以需要非阻塞轮询
        while(1)
        {
            errno = 0;
            InetAddr client;
            int code = 0;
            int sockfd = _listensock->Acceptor(&client, &code);
            if (sockfd > 0)
            {
                // 获取新连接成功
                LOG(INFO, "获取到新连接，客户端信息：[%s]\n", client.AddrStr().c_str());
                // 添加新连接到 Reactor
                con->_R->AddConnection(sockfd, EPOLLIN|EPOLLET, client, NormalSock);
            }
            else
            {
                // 获取连接失败，原因可能有很多. 根据错误码判断
                if (code == EWOULDBLOCK)
                {
                    LOG(INFO, "底层连接全部获取完毕\n");
                    break;
                }
                else if (code == EINTR)
                {
                    // 获取连接被中断
                    continue;
                }
                else
                {
                    LOG(FATAL, "获取连接失败\n");
                    break;
                }
            }
        }
    }
private:
    uint16_t _port;
    std::unique_ptr<Socket> _listensock;
};